<template>
  <div>
    <el-switch
      v-model="draggable"
      active-text="开启拖拽"
      inactive-text="关闭拖拽"
    ></el-switch>
    <el-button v-if="draggable" @click="batchSave">批量保存</el-button>
    <el-tree
      :data="menus"
      :props="defaultProps"
      :expand-on-click-node="false"
      show-checkbox
      node-key="catId"
      :default-expanded-keys="expandedKey"
      :draggable="draggable"
      :allow-drop="allowDrop"
      @node-drop="handleDrop"
    >
      <span class="custom-tree-none" slot-scope="{ node, data }">
        <span>{{ node.label }}</span>
        <span>
          <el-button
            v-if="node.level <= 2"
            type="text"
            sizi="mini"
            @click="() => append(data)"
            >添加</el-button
          >
          <el-button
            v-if="node.childNodes.length == 0"
            type="text"
            sizi="mini"
            @click="() => remove(node, data)"
            >删除</el-button
          >
          <el-button type="text" sizi="mini" @click="() => edit(data)"
            >修改</el-button
          >
        </span>
      </span>
    </el-tree>
    <el-dialog :title="title" :visible.sync="dialogVisible" width="30%">
      <el-form :model="category">
        <el-form-item label="分类名称">
          <el-input v-model="category.name" auto-complete="off"></el-input>
        </el-form-item>
        <el-form-item label="图标">
          <el-input v-model="category.icon" auto-complete="off"></el-input>
        </el-form-item>
        <el-form-item label="计量单位">
          <el-input
            v-model="category.productUnit"
            auto-complete="off"
          ></el-input>
        </el-form-item>
      </el-form>
      <span slot="footer" class="dialog-footer">
        <el-button @click="dialogVisible = false">取 消</el-button>
        <el-button type="primary" @click="submitData">确 定</el-button>
      </span>
    </el-dialog>
  </div>
</template>

<script>
export default {
  data() {
    return {
      draggable: false,
      updateNodes: [],
      title: "",
      dialogType: "", //edit, append
      category: {
        name: "",
        parentCid: 0,
        catLevel: 0,
        showStatus: 1,
        sort: 0,
        catId: null,
        icon: "",
        productUnit: "",
      },
      menus: [],
      defaultProps: {
        children: "children",
        label: "name",
      },
      expandedKey: [],
      dialogVisible: false,
      maxLevel: 0,
      parentId: 0,
    };
  },
  methods: {
    batchSave() {
      //发送网络请求
      this.$http({
        url: this.$http.adornUrl("/product/category/update/sort"),
        method: "post",
        data: this.$http.adornData(this.updateNodes, false),
      }).then(({ data }) => {
        this.$message({
          message: "菜单修改成功",
          type: "success",
        });
        //修改成功，刷新菜单
        this.getMenus();
        //默认展开菜单
        this.expandedKey = [this.parentId];
        this.updateNodes = [];
      });
    },
    handleDrop(draggingNode, dropNode, dropType, ev) {
      //console.log("tree drop: ", dropNode.label, dropType);
      // 1. 当前节点最新的父节点
      let parentId = 0;
      let siblings = null;
      if (dropType == "before" || dropType == "after") {
        //如果是拖拽到了 dropNode 之前或之后，那么拖拽的节点的
        parentId =
          dropNode.parent.data.catId == undefined
            ? 0
            : dropNode.parent.data.catId;

        siblings = dropNode.parent.childNodes;
      } else {
        parentId = dropNode.data.catId;
        siblings = dropNode.childNodes;
      }
      // 2. 当前拖拽节点的最新层数
      for (let i = 0; i < siblings.length; i++) {
        if (siblings[i].data.catId == draggingNode.data.catId) {
          //如果遍历的是当前正在拖拽的节点
          let catLevel = draggingNode.level;
          if (siblings[i].level != draggingNode.level) {
            //如果当前节点的层级发生变化，那么就要获取最新层级
            catLevel = siblings[i].level;
            //修改子节点的层级
            this.updateChildNodeLevel(siblings[i]);
          }
          this.updateNodes.push({
            catId: siblings[i].data.catId,
            sort: i,
            parentCid: parentId,
            catLevel: catLevel,
          });
        } else {
          this.updateNodes.push({ catId: siblings[i].data.catId, sort: i });
        }
      }
      this.parentId = parentId;
      // 3. 当前拖拽节点的最新层级
      //console.log("updateNodes = ", this.updateNodes);
    },
    updateChildNodeLevel(node) {
      if (node.childNodes.length > 0) {
        for (let i = 0; i < node.childNodes.length; i++) {
          var cNode = node.childNodes[i].data;
          this.updateNodes.push({
            catId: cNode.catId,
            catLevel: node.childNodes[i].level,
          });
          this.updateChildNodeLevel(node.childNodes[i]);
        }
      }
    },
    allowDrop(draggingNode, dropNode, type) {
      //console.log(draggingNode, dropNode, type);
      //只需要判断，被拖动的当前节点以及所在的父节点总层数不大于3
      this.maxLevel = 0;
      this.countNodeLevel(draggingNode.data);
      let deep = this.maxLevel - draggingNode.data.catLevel + 1;

      if (type == "inner") {
        deep = deep + dropNode.level;
      } else {
        deep = deep + dropNode.parent.level;
      }
      //console.log("深度 = ", deep);
      return deep <= 3;
    },
    countNodeLevel(node) {
      //1. 找到所有子节点，求出最大深度
      if (node.children != null && node.children.length > 0) {
        for (let i = 0; i < node.children.length; i++) {
          if (node.children[i].catLevel > this.maxLevel) {
            this.maxLevel = node.children[i].catLevel;
          }
          this.countNodeLevel(node.children[i]);
        }
      } else {
        this.maxLevel = node.catLevel;
      }
    },
    clearCategory() {
      this.category.name = "";
      this.category.parentCid = 0;
      this.category.catLevel = 0;
      this.category.showStatus = 1;
      this.category.sort = 0;
      this.category.catId = null;
      this.category.icon = "";
      this.category.productUnit = "";
    },
    handleNodeClick(data) {
      //console.log(data);
    },
    getMenus() {
      this.$http({
        url: this.$http.adornUrl("/product/category/list/tree"),
        method: "get",
      }).then((data) => {
        this.menus = data.data.data;
      });
    },
    edit(data) {
      this.title = "修改分类";
      this.dialogType = "edit";
      this.dialogVisible = true;
      //发送请求获取当前节点最新的数据
      this.$http({
        url: this.$http.adornUrl(`/product/category/info/${data.catId}`),
        method: "get",
      }).then(({ data }) => {
        //console.log("回显的数据：", data);
        this.category.name = data.category.name;
        this.category.catId = data.category.catId;
        this.category.icon = data.category.icon;
        this.category.productUnit = data.category.productUnit;
        this.category.parentCid = data.category.parentCid;
      });
    },
    append(data) {
      this.clearCategory();
      this.title = "添加分类";
      this.dialogType = "append";
      this.dialogVisible = true;
      this.category.parentCid = data.catId;
      this.category.catLevel = parseInt(data.catLevel) + 1;
    },
    submitData() {
      if (this.dialogType == "append") {
        this.addCategory();
      }
      if (this.dialogType == "edit") {
        this.editCategory();
      }
    },
    //添加修改分类数据方法
    editCategory() {
      var { catId, name, icon, productUnit } = this.category;

      this.$http({
        url: this.$http.adornUrl("/product/category/update"),
        method: "post",
        data: this.$http.adornData({ catId, name, icon, productUnit }, false),
      }).then(({ data }) => {
        //关闭对话框
        this.dialogVisible = false;
        //刷新出新的菜单
        this.getMenus();
        //设置需要的展开的菜单
        this.expandedKey = [this.category.parentCid];
        this.$message({
          type: "success",
          message: "菜单保存成功",
        });
      });
    },
    //添加三级分类方法
    addCategory() {
      this.$http({
        url: this.$http.adornUrl("/product/category/save"),
        method: "post",
        data: this.$http.adornData(this.category, false),
      }).then(({ data }) => {
        //关闭对话框
        this.dialogVisible = false;
        //刷新出新的菜单
        this.getMenus();
        //设置需要的展开的菜单
        this.expandedKey = [this.category.parentCid];
        this.$message({
          type: "success",
          message: "菜单保存成功",
        });
      });
    },
    remove(node, data) {
      var ids = [data.catId];
      this.$confirm(`是否删除【${data.name}】菜单?`, "提示", {
        confirmButtonText: "确定",
        cancelButtonText: "取消",
        type: "warning",
      })
        .then(() => {
          this.$http({
            url: this.$http.adornUrl("/product/category/delete"),
            method: "post",
            data: this.$http.adornData(ids, false),
          }).then(({ data }) => {
            //刷新出新的菜单
            this.getMenus();
            //设置需要的展开的菜单
            this.expandedKey = [node.parent.data.catId];
            this.$message({
              message: "删除成功",
              type: "success",
            });
          });
        })
        .catch(() => {
          this.$message({
            type: "info",
            message: "已取消删除",
          });
        });
    },
  },
  created() {
    this.getMenus();
  },
};
</script>


